(draft)feat: add cost estimation per provider for Goose CLI#3312
Closed
GaryZhous wants to merge 27 commits intoblock:mainfrom
Closed
(draft)feat: add cost estimation per provider for Goose CLI#3312GaryZhous wants to merge 27 commits intoblock:mainfrom
GaryZhous wants to merge 27 commits intoblock:mainfrom
Conversation
Co-authored-by: Chaitanya Rahalkar <chaitanyarahalkar4@gmail.com>
Co-authored-by: Angie Jones <jones.angie@gmail.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
…ck#3260) Co-authored-by: dianed-square <73617011+dianed-square@users.noreply.github.com> Co-authored-by: Angie Jones <jones.angie@gmail.com>
Co-authored-by: Rizel Scarlett <rizel@squareup.com>
GaryZhous
force-pushed
the
GaryZ/Refine-CLI
branch
from
c17fe3b to
55869eb
Compare
GaryZhous
changed the title
Contributor
Author
|
this is just a draft |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
13 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Built on top of #2941, this pull request enhances the
goose-clisession module by introducing functionality to estimate and display cost information for token usage based on provider pricing data. It includes new methods for normalizing model names, fetching pricing data, and calculating costs, as well as configuration updates to enable cost display. Instead of hard-coding the cost calculation, I make use of Jack'sPricing_Cachewhich he created in #2992.Pricing Integration:
initialize_pricing_cacheto session startup: Ensures pricing cache is initialized, with error handling to log warnings if initialization fails.estimate_cost_usdfunction: Calculates token usage cost in USD using provider pricing data and normalized model names.display_cost_usagefunction: Displays cost information in USD if pricing data is available, including token breakdown.Model Name Normalization:
normalize_model_namefunction: Cleans up model names by removing suffixes (e.g., "-latest", date-like patterns) and converting version numbers for compatibility with pricing data. Otherwise we are not able to access the pricing info from the cache since Openrouter does the model naming differently. (e.g.claude-3-5-haiku-latestisclaude-3.5-haikuon Openrouter)Configuration Updates:
GOOSE_CLI_SHOW_COSTto toggle cost display andGOOSE_PROVIDERto specify the provider name.These changes collectively improve the
goose-clitool by providing users with detailed cost insights based on their session's token usage. For models whose pricing is unknown, nothing will show up.How to enable
After